#pragma once
#include <iostream>
#include <string>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include "Common.hpp"

class InetAddr
{
public:
    InetAddr(){}

    //Set
    void SetInetAddr(struct sockaddr_in& addr)
    {
        _addr = addr;
        _port = ntohs(_addr.sin_port); // ntohs:net to host short网络转主机short16位
        //_ip = inet_ntoa(_addr.sin_addr);// 不建议使用该接口，公共缓冲区不安全
        char ipbuffer[64]; // 使用自己的缓冲区
        inet_ntop(AF_INET, &_addr.sin_addr, ipbuffer, sizeof(_addr));
        _ip = ipbuffer;
    }

    // 网络转主机
    InetAddr(struct sockaddr_in &addr)
    {
        SetInetAddr(addr);
    }

    // 主机转网络
    InetAddr(const std::string &ip, uint16_t port) : _ip(ip), _port(port)
    {
        memset(&_addr, 0, sizeof(_addr));
        _addr.sin_family = AF_INET;
        _addr.sin_port = htons(_port);
        //_addr.sin_addr.s_addr = inet_addr(_ip.c_str());//同样不建议使用该不安全接口
        inet_pton(AF_INET, _ip.c_str(), &_addr.sin_addr);
    }

    InetAddr(uint16_t port):_port(port),_ip()
    {
        memset(&_addr,0,sizeof(_addr));
        _addr.sin_family = AF_INET;
        _addr.sin_addr.s_addr = INADDR_ANY;
        _addr.sin_port = htons(_port);
    }

    uint16_t Port() { return _port; }
    std::string Ip() { return _ip; }
    const struct sockaddr_in &Addr() { return _addr; }
    const struct sockaddr* NetAddrPtr(){ return CONV(_addr);}
    socklen_t NetAddrLen(){ return sizeof(_addr);}

    std::string StringAddr()
    {
        return _ip + ":" + std::to_string(_port);
    }

    bool operator==(const InetAddr &addr) // 为后续确认唯一用户准备
    {
        return _ip == addr._ip && _port == addr._port;
    }

    ~InetAddr() {}

private:
    struct sockaddr_in _addr; // 网络地址
    std::string _ip;          // 主机地址ip
    uint16_t _port;           // 主机地址port
};